/*
 * Copyright 2017 Google
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#import "FDotInfo.h"
#import "FTestHelpers.h"
#import "FIRDatabaseConfig_Private.h"

@implementation FDotInfo

- (void) testCanGetReferenceToInfoNodes {
    FIRDatabaseReference * ref = [FTestHelpers getRandomNode];

    [ref.root child:@".info"];
    [ref.root child:@".info/foo"];
}

- (void) testCantWriteToInfo {
    FIRDatabaseReference * ref = [[FTestHelpers getRandomNode].root child:@".info"];
    XCTAssertThrows([ref setValue:@"hi"], @"Cannot write to path at /.info");
    XCTAssertThrows([ref setValue:@"hi" andPriority:@5], @"Cannot write to path at /.info");
    XCTAssertThrows([ref setPriority:@"hi"], @"Cannot write to path at /.info");
    XCTAssertThrows([ref runTransactionBlock:^FIRTransactionResult *(FIRMutableData *currentData) {
        return [FIRTransactionResult successWithValue:currentData];
    }], @"Cannot write to path at /.info");
    XCTAssertThrows([ref removeValue], @"Cannot write to path at /.info");
    XCTAssertThrows([[ref child:@"test"] setValue:@"hi"], @"Cannot write to path at /.info");
}

- (void) testCanWatchInfoConnected {
    FIRDatabaseReference * rootRef = [FTestHelpers getRandomNode].root;
    __block BOOL done = NO;
    [[rootRef child:@".info/connected"] observeEventType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        if ([[snapshot value] boolValue]) {
            done = YES;
        }
    }];
    [self waitUntil:^{ return done; }];
}

- (void) testInfoConnectedGoesToFalseOnDisconnect {
    FIRDatabaseConfig *cfg = [FIRDatabaseConfig configForName:@"test-config"];
    FIRDatabaseReference * rootRef = [[FIRDatabaseReference alloc] initWithConfig:cfg];
    __block BOOL everConnected = NO;
    __block NSMutableString *connectedHistory = [[NSMutableString alloc] init];
    [[rootRef child:@".info/connected"] observeEventType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        if ([[snapshot value] boolValue]) {
            everConnected = YES;
        }

        if (everConnected) {
            [connectedHistory appendString:([[snapshot value] boolValue] ? @"YES," : @"NO,")];
        }
    }];
    [self waitUntil:^{ return everConnected; }];

    [FRepoManager interrupt:cfg];
    [FRepoManager resume:cfg];

    [self waitUntil:^BOOL{
        return [connectedHistory isEqualToString:@"YES,NO,YES,"];
    }];

    [FRepoManager interrupt:cfg];
    [FRepoManager disposeRepos:cfg];
}

- (void) testInfoServerTimeOffset {
    FIRDatabaseConfig *cfg = [FIRDatabaseConfig configForName:@"test-config"];
    FIRDatabaseReference * ref = [[FIRDatabaseReference alloc] initWithConfig:cfg];

    // make sure childByAutoId works
    [ref childByAutoId];

    NSMutableArray* offsets = [[NSMutableArray alloc] init];

    [[ref child:@".info/serverTimeOffset"] observeEventType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        NSLog(@"got value: %@", snapshot.value);
        [offsets addObject:snapshot.value];
    }];

    WAIT_FOR(offsets.count == 1);

    XCTAssertTrue([[offsets objectAtIndex:0] isKindOfClass:[NSNumber class]], @"Second element should be a number, in milliseconds");

    // make sure childByAutoId still works
    [ref childByAutoId];

    [FRepoManager interrupt:cfg];
    [FRepoManager disposeRepos:cfg];
}

- (void) testManualConnectionManagement {
    FIRDatabaseConfig *cfg = [FIRDatabaseConfig configForName:@"test-config"];
    FIRDatabaseConfig *altCfg = [FIRDatabaseConfig configForName:@"alt-config"];

    FIRDatabaseReference * ref = [[FIRDatabaseReference alloc] initWithConfig:cfg];
    FIRDatabaseReference * refAlt = [[FIRDatabaseReference alloc] initWithConfig:altCfg];

    // Wait until we're connected to both Firebases
    __block BOOL ready = NO;
    [[ref child:@".info/connected"] observeEventType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        ready = [[snapshot value] boolValue];
    }];
    [self waitUntil:^{ return ready; }];
    [[ref child:@".info/connected"] removeAllObservers];

    ready = NO;
    [[refAlt child:@".info/connected"] observeEventType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        ready = [[snapshot value] boolValue];
    }];
    [self waitUntil:^{ return ready; }];
    [[refAlt child:@".info/connected"] removeAllObservers];

    [FIRDatabaseReference goOffline];

    // Ensure we're disconnected from both Firebases
    ready = NO;

    [[ref child:@".info/connected"] observeSingleEventOfType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        XCTAssertFalse([[snapshot value] boolValue], @".info/connected should be false");
        ready = YES;
    }];
    [self waitUntil:^{ return ready; }];
    ready = NO;
    [[refAlt child:@".info/connected"] observeSingleEventOfType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        XCTAssertFalse([[snapshot value] boolValue], @".info/connected should be false");
        ready = YES;
    }];
    [self waitUntil:^{ return ready; }];

    // Ensure that we don't automatically reconnect upon new Firebase creation
    FIRDatabaseReference * refDup = [[FIRDatabaseReference alloc] initWithConfig:altCfg];
    [[refDup child:@".info/connected"] observeEventType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        if ([[snapshot value] boolValue]) {
            XCTFail(@".info/connected should remain false");
        }
    }];

    // Wait for 1.5 seconds to make sure connected remains false
    [NSThread sleepForTimeInterval:1.5];
    [[refDup child:@".info/connected"] removeAllObservers];

    [FIRDatabaseReference goOnline];

    // Ensure we're reconnected to both Firebases
    ready = NO;
    [[ref child:@".info/connected"] observeEventType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        ready = [[snapshot value] boolValue];
    }];
    [self waitUntil:^{ return ready; }];
    [[ref child:@".info/connected"] removeAllObservers];

    ready = NO;
    [[refAlt child:@".info/connected"] observeEventType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot *snapshot) {
        ready = [[snapshot value] boolValue];
    }];
    [self waitUntil:^{ return ready; }];
    [[refAlt child:@".info/connected"] removeAllObservers];
}
@end
